{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import numpy as np\n",
    "\n",
    "import os\n",
    "from pathlib import Path"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "data_dir = \"../data/raw/\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "cwd = Path(os.curdir)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "data_dir = cwd / 'data' / 'raw'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "file_dir = Path().cwd()\n",
    "# uncomment when using in a .py file\n",
    "# file_dir = Path(__file__)\n",
    "base_dir = file_dir.parents[0]\n",
    "data_dir = base_dir / 'data'\n",
    "raw_data_dir = data_dir / 'raw' / 'lmd-full_and_reddit_MIDI_dataset'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "sentence_level_31 = raw_data_dir / 'sentence_level_31'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [],
   "source": [
    "fnames = sentence_level_31.iterdir()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "outputs": [],
   "source": [],
   "metadata": {
    "collapsed": false,
    "pycharm": {
     "name": "#%%\n"
    }
   }
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [],
   "source": [
    "count = 0\n",
    "for i in fnames:\n",
    "#     print(i)\n",
    "#     data = np.load(i)\n",
    "#     data = data[0]\n",
    "    count+=1\n",
    "#     print(data[0][0][0])\n",
    "#     print(len(data[0]))\n",
    "#     print(data[1][0][0])\n",
    "#     print(len(data[0]))\n",
    "#     print(data[2][0][0])\n",
    "#     print(len(data[0]))\n",
    "    break"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": "1"
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "count"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 71,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(3,)"
      ]
     },
     "execution_count": 71,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data.shape"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 72,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "21"
      ]
     },
     "execution_count": 72,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "len(data[0])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 73,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "21"
      ]
     },
     "execution_count": 73,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "len(data[0])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 74,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[[[0.0, 0.324998916666666, 415.3046975799451, 95.0],\n",
       "  [0.4339271249999994, 0.36071308333333363, 277.1826309768721, 83.0],\n",
       "  [0.5178554166666673, 0.24107062499999987, 311.1269837220809, 83.0],\n",
       "  [0.33035604166666577, 0.17857083333333357, 329.6275569128699, 79.0],\n",
       "  [0.3482131250000009, 0.35714166666666536, 369.9944227116344, 80.0],\n",
       "  [0.5178554166666665, 0.19642791666666604, 466.1637615180899, 73.0],\n",
       "  [0.3482131249999991, 0.17857083333333357, 369.9944227116344, 83.0],\n",
       "  [0.3249989166666669, 0.18988037500000132, 311.1269837220809, 75.0]],\n",
       " [[1.5589266250000016, 0.3412694999999992, 415.3046975799451, 73.0],\n",
       "  [0.476189999999999, 0.2579362500000002, 277.1826309768721, 73.0],\n",
       "  [0.4563487500000001, 0.24801562500000074, 311.1269837220809, 69.0],\n",
       "  [0.3472218750000007, 0.23214262499999982, 329.6275569128699, 79.0],\n",
       "  [0.42261862500000014, 0.4067456249999992, 369.9944227116344, 65.0],\n",
       "  [0.555555, 0.327380625, 466.1637615180899, 67.0],\n",
       "  [0.5059518749999992, 0.962300625000001, 440.0, 80.0]],\n",
       " [[2.946425625, 0.4305551250000015, 415.3046975799451, 90.0],\n",
       "  [0.569443875000001, 0.2777774999999991, 277.1826309768721, 71.0],\n",
       "  [0.4067456249999992, 0.23809500000000128, 311.1269837220809, 64.0],\n",
       "  [0.3630948750000016, 0.43055512499999793, 329.6275569128699, 92.0],\n",
       "  [0.579364499999997, 0.271825125000003, 369.9944227116344, 90.0],\n",
       "  [0.5019836250000012, 0.32738062500000353, 554.3652619537442, 60.0],\n",
       "  [0.410713874999999, 0.337301250000003, 493.8833012561241, 84.0],\n",
       "  [0.4365075000000047, 0.29761874999999804, 369.9944227116344, 73.0]],\n",
       " [[1.5634905000000003, 0.48015824999999523, 329.6275569128699, 80.0],\n",
       "  [0.5257931249999999, 0.4067456249999992, 493.8833012561241, 72.0],\n",
       "  [0.48015824999999523, 0.5059518750000009, 415.3046975799451, 90.0],\n",
       "  [0.5317455000000031, 0.5496026250000021, 329.6275569128699, 79.0],\n",
       "  [0.6051581249999991, 1.0143796250000037, 311.1269837220809, 65.0]],\n",
       " [[2.534444270833337, 0.5159047916666637, 415.3046975799451, 75.0],\n",
       "  [0.5340864583333307, 0.2999974999999999, 277.1826309768721, 57.0],\n",
       "  [0.43181458333333467, 0.39090583333333484, 311.1269837220809, 61.0],\n",
       "  [0.46590520833333215, 0.4477235416666687, 329.6275569128699, 90.0],\n",
       "  [0.5340864583333342, 0.504541249999999, 369.9944227116344, 95.0],\n",
       "  [0.6363583333333338, 0.470450624999998, 466.1637615180899, 83.0],\n",
       "  [0.5568135416666671, 0.2999974999999999, 369.9944227116344, 95.0],\n",
       "  [0.34090624999999974, 0.3568152083333338, 311.1269837220809, 90.0]],\n",
       " [[1.0340822916666674, 0.36817874999999844, 415.3046975799451, 80.0],\n",
       "  [0.3977239583333336, 0.2954520833333305, 277.1826309768721, 69.0],\n",
       "  [0.45908708333333337, 0.4318145833333311, 311.1269837220809, 80.0],\n",
       "  [0.49545041666666734, 0.5159047916666637, 329.6275569128699, 82.0],\n",
       "  [0.6068131250000022, 0.590904166666661, 369.9944227116344, 90.0],\n",
       "  [0.6772670833333336, 0.41363291666666413, 466.1637615180899, 95.0],\n",
       "  [0.47726874999999325, 1.2079885416666727, 440.0, 72.0]],\n",
       " [[2.065022791666671, 0.27727041666665997, 415.3046975799451, 100.0],\n",
       "  [0.3522697916666644, 0.2545433333333307, 277.1826309768721, 73.0],\n",
       "  [0.3977239583333301, 0.32272458333333276, 311.1269837220809, 77.0],\n",
       "  [0.4590870833333369, 0.5272683333333319, 329.6275569128699, 97.0],\n",
       "  [0.6818124999999995, 0.3522697916666644, 369.9944227116344, 95.0],\n",
       "  [0.5749952083333341, 0.11818083333333362, 493.8833012561241, 66.0],\n",
       "  [0.40226937499999593, 0.348294166666669, 493.8833012561241, 90.0],\n",
       "  [0.4357941666666676, 0.59375, 369.9944227116344, 86.0]],\n",
       " [[1.421875, 0.4750000000000014, 329.6275569128699, 90.0],\n",
       "  [0.5218750000000014, 0.453125, 493.8833012561241, 95.0],\n",
       "  [0.4781249999999986, 0.734375, 415.3046975799451, 86.0],\n",
       "  [0.7406250000000014, 0.9125000000000014, 329.6275569128699, 92.0],\n",
       "  [3.125, 0.7562500000000014, 369.9944227116344, 90.0],\n",
       "  [0.78125, 0.8846089374999977, 311.1269837220809, 64.0],\n",
       "  [1.1307609374999998, 0.2884593750000022, 466.1637615180899, 73.0],\n",
       "  [0.3942278124999987, 1.321143937499997, 493.8833012561241, 90.0],\n",
       "  [2.192291250000004, 0.1576911249999995, 311.1269837220809, 71.0],\n",
       "  [0.28269018749999475, 1.048069062500005, 329.6275569128699, 74.0]],\n",
       " [[1.0922995000000029, 0.2442289374999973, 466.1637615180899, 77.0],\n",
       "  [0.36922799999999967, 1.4365276874999964, 493.8833012561241, 95.0]],\n",
       " [[1.7269101249999963, 0.19615237500000404, 329.6275569128699, 66.0],\n",
       "  [0.34615125000000546, 0.8307629999999975, 369.9944227116344, 95.0],\n",
       "  [1.1634528124999974, 0.19230625000000146, 466.1637615180899, 75.0],\n",
       "  [0.3115361249999964, 0.2211521875000031, 493.8833012561241, 97.0],\n",
       "  [0.307690000000008, 0.8365321874999978, 391.99543598174927, 88.0],\n",
       "  [1.0249923124999967, 0.3076900000000009, 466.1637615180899, 90.0],\n",
       "  [0.41345843749999744, 0.28269018750000185, 493.8833012561241, 79.0],\n",
       "  [0.3749971875, 0.6826871875000009, 415.3046975799451, 90.0],\n",
       "  [0.7730711250000013, 0.4076892499999971, 466.1637615180899, 80.0],\n",
       "  [0.48076562499999653, 1.2057601875000046, 493.8833012561241, 92.0]],\n",
       " [[2.0442154375, 0.3211514375000064, 466.1637615180899, 106.0],\n",
       "  [0.3692280000000068, 0.2403828124999947, 493.8833012561241, 82.0],\n",
       "  [1.7942173124999954, 0.09615312500000073, 415.3046975799451, 64.0],\n",
       "  [0.6096108125000015, 0.7826864374999971, 415.3046975799451, 95.0],\n",
       "  [1.1634528125000045, 0.5384574999999927, 369.9944227116344, 90.0],\n",
       "  [0.6980716875000041, 0.34038206249999803, 369.9944227116344, 86.0],\n",
       "  [0.8884548749999936, 0.26345956250000313, 349.2282314330039, 77.0],\n",
       "  [0.40384312499999453, 0.2980746875000051, 369.9944227116344, 75.0],\n",
       "  [0.38076637500000743, 0.6192261249999973, 277.1826309768721, 86.0],\n",
       "  [1.0769149999999996, 0.32692062499999963, 311.1269837220809, 90.0]],\n",
       " [[8.676857999999996, 0.1442296875000011, 493.8833012561241, 97.0],\n",
       "  [0.25961343750000765, 0.1442296875000011, 493.8833012561241, 84.0],\n",
       "  [0.32307449999998994, 1.6576798750000137, 493.8833012561241, 103.0],\n",
       "  [2.865363125000002, 0.11922987500000204, 246.94165062806206, 84.0],\n",
       "  [0.3942278125000058, 0.221152187499996, 493.8833012561241, 109.0],\n",
       "  [0.3115361250000035, 0.12499906249999526, 493.8833012561241, 103.0],\n",
       "  [0.3038438749999841, 1.6384492500000079, 493.8833012561241, 97.0]],\n",
       " [[2.038446250000007, 0.8269168750000091, 369.9944227116344, 112.0],\n",
       "  [0.9903771875000018, 0.2211521875000102, 466.1637615180899, 100.0],\n",
       "  [0.32692062499999963, 0.18653706250000823, 493.8833012561241, 97.0],\n",
       "  [0.4076892500000042, 0.8076862499999891, 391.99543598174927, 126.0],\n",
       "  [0.9769157499999892, 0.29230549999999766, 466.1637615180899, 95.0],\n",
       "  [0.40384312500000874, 0.503842375000005, 493.8833012561241, 95.0],\n",
       "  [0.6576873749999947, 0.2980746875000051, 415.3046975799451, 106.0],\n",
       "  [0.3942278125000058, 0.21153687499999307, 466.1637615180899, 100.0],\n",
       "  [0.3615357499999874, 1.9422931250000204, 493.8833012561241, 113.0]],\n",
       " [[2.455750812500014, 0.34038206249999803, 466.1637615180899, 126.0],\n",
       "  [0.4326890625000033, 0.33653593750000255, 493.8833012561241, 117.0],\n",
       "  [1.6884488749999917, 0.13461437499999818, 415.3046975799451, 100.0],\n",
       "  [0.7346098750000039, 0.3749971875, 415.3046975799451, 106.0],\n",
       "  [0.553842000000003, 0.3211514374999922, 369.9944227116344, 112.0],\n",
       "  [0.34615124999999125, 0.28269018750000896, 349.2282314330039, 71.0],\n",
       "  [0.3692280000000068, 0.7307637499999942, 369.9944227116344, 118.0],\n",
       "  [0.7403790624999971, 1.5711420625000017, 277.1826309768721, 86.0],\n",
       "  [2.79805593750001, 0.5961493749999818, 246.94165062806206, 117.0],\n",
       "  [0.7115331249999883, 2.5615192500000035, 415.3046975799451, 118.0]],\n",
       " [[8.063401062499992, 0.28269018750000896, 466.1637615180899, 118.0],\n",
       "  [0.4326890625000033, 0.34038206249999803, 415.3046975799451, 118.0],\n",
       "  [0.3749971875000142, 0.869224250000002, 329.6275569128699, 109.0],\n",
       "  [1.028838437499985, 0.26345956250000313, 466.1637615180899, 126.0],\n",
       "  [0.3211514375000064, 0.18653706249999402, 415.3046975799451, 112.0],\n",
       "  [0.2980746875000051, 0.2307674999999989, 329.6275569128699, 109.0],\n",
       "  [0.32307449999998994, 1.0807611250000093, 329.6275569128699, 97.0]],\n",
       " [[4.163430312499997, 0.5807648749999998, 466.1637615180899, 118.0],\n",
       "  [0.5865340625000215, 0.2980746874999909, 415.3046975799451, 112.0],\n",
       "  [0.3942278124999916, 1.1249915624999858, 311.1269837220809, 112.0],\n",
       "  [1.2922979999999882, 0.3211514375000206, 466.1637615180899, 118.0],\n",
       "  [0.3557665625000084, 0.3557665625000084, 415.3046975799451, 106.0],\n",
       "  [0.38076637500000743, 1.0480690624999909, 311.1269837220809, 90.0]],\n",
       " [[4.644195937500001, 0.35576656249997995, 493.8833012561241, 106.0],\n",
       "  [0.3980739374999871, 0.2346136250000086, 466.1637615180899, 100.0],\n",
       "  [0.3711510624999903, 0.8846087500000124, 369.9944227116344, 100.0],\n",
       "  [1.2596059375000266, 0.2634595624999747, 493.8833012561241, 113.0],\n",
       "  [0.3211514374999922, 0.1538449999999898, 466.1637615180899, 106.0],\n",
       "  [0.2980746874999909, 0.20576768750001406, 369.9944227116344, 106.0],\n",
       "  [0.32307450000001836, 1.455758312499995, 369.9944227116344, 112.0]],\n",
       " [[2.3307517499999904, 0.3211514374999922, 415.3046975799451, 112.0],\n",
       "  [0.4576888750000023, 0.1769217499999911, 466.1637615180899, 112.0],\n",
       "  [0.26922874999999635, 0.17307562499999563, 493.8833012561241, 95.0],\n",
       "  [0.3115361250000035, 1.7499868749999905, 554.3652619537442, 118.0],\n",
       "  [1.8134479375000012, 1.8596014375000038, 523.2511306011972, 109.0],\n",
       "  [1.9076780000000042, 3.432666562499975, 329.6275569128699, 90.0]],\n",
       " [[5.77495668749998, 1.5326808125000184, 329.6275569128699, 122.0],\n",
       "  [1.696141124999997, 1.17691425000001, 329.6275569128699, 113.0],\n",
       "  [1.528834687500023, 1.6576798749999853, 523.2511306011972, 126.0],\n",
       "  [1.82306324999999, 0.21538300000000277, 523.2511306011972, 100.0],\n",
       "  [0.3653818749999971, 1.090376437499998, 523.2511306011972, 126.0],\n",
       "  [1.1730681250000146, 0.2307674999999847, 523.2511306011972, 95.0],\n",
       "  [0.3173053124999967, 1.4076817499999947, 523.2511306011972, 100.0],\n",
       "  [1.7499868749999905, 0.1576911250000137, 349.2282314330039, 97.0]],\n",
       " [[0.32692062500001384, 1.0711458124999922, 391.99543598174927, 127.0],\n",
       "  [1.3846049999999934, 0.3211514374999922, 523.2511306011972, 113.0],\n",
       "  [0.4711503124999865, 0.9038393750000182, 415.3046975799451, 118.0],\n",
       "  [1.1961448750000159, 0.2596134374999792, 493.8833012561241, 109.0],\n",
       "  [0.34230512499999577, 0.3307667499999809, 523.2511306011972, 117.0],\n",
       "  [0.4519196875000091, 1.2788365624999756, 659.2551138257398, 117.0],\n",
       "  [1.721140937499996, 1.6368403124999702, 622.2539674441618, 126.0],\n",
       "  [1.6223055208333221, 0.3691837083333098, 587.3295358348151, 106.0],\n",
       "  [0.43604374999998186, 1.5686378125000147, 523.2511306011972, 109.0]],\n",
       " [[1.9532503125000176, 0.10961456249998491, 440.0, 103.0],\n",
       "  [0.6538412499999993, 0.442304374999992, 440.0, 122.0],\n",
       "  [0.5576881249999985, 0.28269018749998054, 391.99543598174927, 113.0],\n",
       "  [0.3846125000000029, 0.28269018749998054, 369.9944227116344, 100.0],\n",
       "  [0.41345843749999744, 0.8499936249999962, 391.99543598174927, 109.0],\n",
       "  [0.8980701874999966, 2.2730598750000013, 587.3295358348151, 122.0],\n",
       "  [2.692287499999992, 0.11977781249998998, 587.3295358348151, 117.0],\n",
       "  [1.6604551041666582, 2.67043504166665, 523.2511306011972, 127.0]]]"
      ]
     },
     "execution_count": 74,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data[0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 75,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[[[68.0, 1.0, 0.0],\n",
       "  [61.0, 1.5, 0.0],\n",
       "  [63.0, 0.5, 0.0],\n",
       "  [64.0, 1.0, 0.0],\n",
       "  [66.0, 1.5, 0.0],\n",
       "  [70.0, 1.0, 0.0],\n",
       "  [66.0, 1.0, 0.0],\n",
       "  [63.0, 0.5, 0.0]],\n",
       " [[68.0, 1.0, 4.0],\n",
       "  [61.0, 1.0, 0.0],\n",
       "  [63.0, 1.0, 0.0],\n",
       "  [64.0, 1.0, 0.0],\n",
       "  [66.0, 1.5, 0.0],\n",
       "  [70.0, 1.5, 0.0],\n",
       "  [69.0, 3.0, 0.0]],\n",
       " [[68.0, 1.0, 4.0],\n",
       "  [61.0, 1.0, 0.0],\n",
       "  [63.0, 1.0, 0.0],\n",
       "  [64.0, 1.5, 0.0],\n",
       "  [66.0, 1.0, 0.0],\n",
       "  [73.0, 1.0, 0.0],\n",
       "  [71.0, 1.5, 0.0],\n",
       "  [66.0, 1.0, 0.0]],\n",
       " [[64.0, 1.0, 4.0],\n",
       "  [71.0, 1.0, 0.0],\n",
       "  [68.0, 1.0, 0.0],\n",
       "  [64.0, 1.0, 0.0],\n",
       "  [63.0, 3.0, 0.0]],\n",
       " [[68.0, 1.0, 4.0],\n",
       "  [61.0, 1.5, 0.0],\n",
       "  [63.0, 1.0, 0.0],\n",
       "  [64.0, 1.0, 0.0],\n",
       "  [66.0, 1.5, 0.0],\n",
       "  [70.0, 1.0, 0.0],\n",
       "  [66.0, 1.0, 0.0],\n",
       "  [63.0, 1.0, 0.0]],\n",
       " [[68.0, 1.0, 2.0],\n",
       "  [61.0, 1.5, 0.0],\n",
       "  [63.0, 1.0, 0.0],\n",
       "  [64.0, 1.0, 0.0],\n",
       "  [66.0, 2.0, 0.0],\n",
       "  [70.0, 1.0, 0.0],\n",
       "  [69.0, 3.0, 0.0]],\n",
       " [[68.0, 0.5, 2.0],\n",
       "  [61.0, 1.0, 0.0],\n",
       "  [63.0, 1.5, 0.0],\n",
       "  [64.0, 1.5, 0.0],\n",
       "  [66.0, 1.5, 0.0],\n",
       "  [71.0, 0.75, 0.0],\n",
       "  [71.0, 1.0, 0.0],\n",
       "  [66.0, 2.0, 0.0]],\n",
       " [[64.0, 1.0, 2.0],\n",
       "  [71.0, 1.0, 0.0],\n",
       "  [68.0, 2.0, 0.0],\n",
       "  [64.0, 2.0, 0.0],\n",
       "  [66.0, 2.0, 4.0],\n",
       "  [63.0, 2.5, 0.0],\n",
       "  [70.0, 1.5, 0.0],\n",
       "  [71.0, 3.0, 0.0],\n",
       "  [63.0, 1.0, 2.0],\n",
       "  [64.0, 3.0, 0.0]],\n",
       " [[70.0, 0.5, 0.0], [71.0, 4.0, 0.0]],\n",
       " [[64.0, 0.5, 1.0],\n",
       "  [66.0, 2.0, 0.0],\n",
       "  [70.0, 1.0, 1.0],\n",
       "  [71.0, 0.5, 0.0],\n",
       "  [67.0, 2.5, 0.0],\n",
       "  [70.0, 1.5, 0.0],\n",
       "  [71.0, 0.5, 0.0],\n",
       "  [68.0, 2.0, 0.0],\n",
       "  [70.0, 1.0, 0.0],\n",
       "  [71.0, 3.0, 0.0]],\n",
       " [[70.0, 1.0, 2.0],\n",
       "  [71.0, 0.5, 0.0],\n",
       "  [68.0, 0.25, 4.0],\n",
       "  [68.0, 2.0, 1.0],\n",
       "  [66.0, 1.5, 1.0],\n",
       "  [66.0, 1.0, 0.0],\n",
       "  [65.0, 1.0, 1.0],\n",
       "  [66.0, 1.0, 0.0],\n",
       "  [61.0, 2.0, 0.0],\n",
       "  [63.0, 1.0, 1.0]],\n",
       " [[71.0, 0.5, 16.0],\n",
       "  [71.0, 1.0, 0.0],\n",
       "  [71.0, 4.0, 0.0],\n",
       "  [59.0, 0.75, 4.0],\n",
       "  [71.0, 0.5, 0.0],\n",
       "  [71.0, 0.75, 0.0],\n",
       "  [71.0, 4.0, 0.0]],\n",
       " [[66.0, 2.0, 1.0],\n",
       "  [70.0, 1.0, 0.0],\n",
       "  [71.0, 1.0, 0.0],\n",
       "  [67.0, 2.5, 0.0],\n",
       "  [70.0, 1.5, 0.0],\n",
       "  [71.0, 1.5, 0.0],\n",
       "  [68.0, 1.5, 0.0],\n",
       "  [70.0, 1.0, 0.0],\n",
       "  [71.0, 6.0, 0.0]],\n",
       " [[70.0, 1.0, 1.0],\n",
       "  [71.0, 1.0, 0.0],\n",
       "  [68.0, 0.25, 4.0],\n",
       "  [68.0, 1.5, 2.0],\n",
       "  [66.0, 1.0, 0.0],\n",
       "  [65.0, 0.5, 0.0],\n",
       "  [66.0, 2.0, 0.0],\n",
       "  [61.0, 4.0, 0.0],\n",
       "  [59.0, 2.5, 4.0],\n",
       "  [68.0, 6.0, 0.0]],\n",
       " [[70.0, 0.5, 16.0],\n",
       "  [68.0, 1.0, 0.0],\n",
       "  [64.0, 2.5, 0.0],\n",
       "  [70.0, 0.5, 0.0],\n",
       "  [68.0, 1.0, 0.0],\n",
       "  [64.0, 0.5, 0.0],\n",
       "  [64.0, 3.0, 0.0]],\n",
       " [[70.0, 2.0, 8.0],\n",
       "  [68.0, 1.5, 0.0],\n",
       "  [63.0, 3.5, 0.0],\n",
       "  [70.0, 1.0, 0.0],\n",
       "  [68.0, 1.0, 0.0],\n",
       "  [63.0, 3.0, 0.0]],\n",
       " [[71.0, 1.0, 8.0],\n",
       "  [70.0, 1.0, 0.0],\n",
       "  [66.0, 2.0, 0.0],\n",
       "  [71.0, 0.5, 1.0],\n",
       "  [70.0, 1.0, 0.0],\n",
       "  [66.0, 1.0, 0.0],\n",
       "  [66.0, 4.0, 0.0]],\n",
       " [[68.0, 1.0, 2.0],\n",
       "  [70.0, 0.5, 0.0],\n",
       "  [71.0, 1.0, 0.0],\n",
       "  [73.0, 4.0, 0.0],\n",
       "  [72.0, 4.0, 0.0],\n",
       "  [64.0, 8.0, 0.0]],\n",
       " [[64.0, 4.0, 8.0],\n",
       "  [64.0, 3.0, 0.0],\n",
       "  [72.0, 4.5, 1.0],\n",
       "  [72.0, 1.0, 0.0],\n",
       "  [72.0, 3.0, 0.0],\n",
       "  [72.0, 0.5, 0.0],\n",
       "  [72.0, 4.0, 0.0],\n",
       "  [65.0, 1.0, 1.0]],\n",
       " [[67.0, 3.0, 0.0],\n",
       "  [72.0, 1.5, 1.0],\n",
       "  [68.0, 2.0, 0.0],\n",
       "  [71.0, 0.5, 1.0],\n",
       "  [72.0, 1.5, 0.0],\n",
       "  [76.0, 3.0, 0.0],\n",
       "  [75.0, 4.0, 1.0],\n",
       "  [74.0, 1.0, 0.0],\n",
       "  [72.0, 4.0, 0.0]],\n",
       " [[69.0, 0.25, 1.0],\n",
       "  [69.0, 1.5, 1.0],\n",
       "  [67.0, 1.0, 0.0],\n",
       "  [66.0, 1.0, 0.0],\n",
       "  [67.0, 2.0, 0.0],\n",
       "  [74.0, 6.0, 0.0],\n",
       "  [74.0, 0.25, 1.0],\n",
       "  [72.0, 6.0, 4.0]]]"
      ]
     },
     "execution_count": 75,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data[1]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 76,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[[\"She's\", 'a', 'fool', 'and', \"don't\", 'I', 'know', 'it'],\n",
       " ['But', 'a', 'fool', 'can', 'have', 'her', 'charms'],\n",
       " [\"I'm\", 'in', 'love', 'and', \"don't\", 'I', 'show', 'it'],\n",
       " ['Like', 'a', 'babe', 'in', 'arms'],\n",
       " [\"Love's\", 'the', 'same', 'old', 'sad', 'sen', 'sa', 'tion'],\n",
       " ['Late', 'ly', \"I've\", 'not', 'slept', 'a', 'wink'],\n",
       " ['Since', 'this', 'half', 'pint', 'im', 'i', 'ta', 'tion'],\n",
       " ['Put', 'me', 'on', 'the', \"I'm\", 'wild', 'a', 'gain', 'be', 'guiled'],\n",
       " ['a', 'gain'],\n",
       " ['A', 'sim', 'per', 'ing', 'whim', 'per', 'ing', 'child', 'a', 'gain'],\n",
       " ['Be', 'witched', 'both', 'ered', 'and', 'be', 'wil', 'dered', 'am', 'I'],\n",
       " ['Could', \"n't\", 'sleep', 'and', 'would', \"n't\", 'sleep'],\n",
       " ['Love', 'came', 'and', 'told', 'me', 'I', 'should', \"n't\", 'sleep'],\n",
       " ['Be', 'witched', 'both', 'ered', 'and', 'be', 'wil', 'dered', 'am', 'I'],\n",
       " ['Lost', 'my', 'heart', 'but', 'what', 'of', 'it?'],\n",
       " ['She', 'is', 'cold', 'I', 'a', 'gree'],\n",
       " ['She', 'can', 'laugh', 'but', 'I', 'love', 'it'],\n",
       " ['Al', 'though', 'the', \"laugh's\", 'on', 'me'],\n",
       " [\"I'll\", 'sing', 'to', 'her', 'bring', 'spring', 'to', 'her'],\n",
       " ['And', 'long', 'for', 'the', 'day', 'when', \"I'll\", 'to', 'her'],\n",
       " ['Be', 'witched', 'both', 'ered', 'and', 'be', 'dered', 'am']]"
      ]
     },
     "execution_count": 76,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data[2]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.6.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}